python create array of size n